# coding=utf-8
import numpy as np

us_data = "../US_video_data_numbers.csv"
uk_data = "../GB_video_data_numbers.csv"

# 加载国家数据
us_data = np.loadtxt(us_data, delimiter=",", dtype=int)
uk_data = np.loadtxt(uk_data, delimiter=",", dtype=int)

# 添加国家信息
# 分别构造和原来数据相同行数的全为0和全为1的数组分别代表(行拼接)
zeros_data = np.zeros((us_data.shape[0], 1)).astype(int)
ones_data = np.ones((uk_data.shape[0], 1)).astype(int)

# 分别添加一列全为0，1的数组(构造数据)
us_data = np.hstack((us_data, zeros_data))
uk_data = np.hstack((uk_data, ones_data))
print(us_data)
print('*'*50)
# 拼接两组数据(列拼接)
final_data = np.vstack((us_data, uk_data))
print(final_data)
